package com.lishem.carl._10dp;

/**
 * https://leetcode.cn/problems/last-stone-weight-ii/description/
 * <p>
 * 有一堆石头，用整数数组stones表示。其中stones[i]表示第i块石头的重量。
 * <p>
 * 每一回合，从中选出任意两块石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且x<=y。那么粉碎的可能结果如下：
 * <p>
 * 如果 x == y，那么两块石头都会被完全粉碎；
 * <p>
 * 如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。
 * <p>
 * 最后，最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下，就返回 0。
 * <p>
 * 思路：分析案例和结果，转换为 01背包问题
 */
public class _10LetCode1049_最后一块石头重量2 {

    /**
     * 二维数组版本
     */
    public int lastStoneWeightII(int[] stones) {
        int sum = 0;
        for (int stone : stones) {
            sum += stone;
        }
        int target = sum / 2;
        int[][] dp = new int[stones.length][target + 1];
        for (int i = 0; i < dp[0].length; i++) {
            dp[0][i] = i >= stones[0] ? stones[0] : 0;
        }
        for (int i = 1; i < stones.length; i++) {
            for (int j = 0; j < dp[i].length; j++) {
                if (j >= stones[i]) {
                    dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - stones[i]] + stones[i]);
                } else {
                    dp[i][j] = dp[i - 1][j];
                }
            }
        }
        return sum - 2 * dp[stones.length - 1][target];
    }

    /**
     * 滚动数组版本
     */
    public int lastStoneWeightII1(int[] stones) {
        int sum = 0;
        for (int stone : stones) {
            sum += stone;
        }
        int target = sum / 2;
        int[] dp = new int[target + 1];

        for (int i = 0; i < stones.length; i++) {
            for (int j = dp.length - 1; j >= 0; j--) {
                if (j >= stones[i]) {
                    dp[j] = Math.max(dp[j], dp[j - stones[i]] + stones[i]);
                }
            }
        }
        return sum - 2 * dp[target];
    }

    public static void main(String[] args) {
        _10LetCode1049_最后一块石头重量2 sol = new _10LetCode1049_最后一块石头重量2();
        System.out.println(sol.lastStoneWeightII(new int[]{2, 7, 4, 1, 8, 1}));
        System.out.println(sol.lastStoneWeightII1(new int[]{2, 7, 4, 1, 8, 1}));

        System.out.println(sol.lastStoneWeightII(new int[]{31, 26, 33, 21, 40}));
        System.out.println(sol.lastStoneWeightII1(new int[]{31, 26, 33, 21, 40}));
    }
}